import fs from "fs";

export const fileLoggingHandler = {
  // [ 'Human: 你好', 'AI: xxxx' ]
  handleLLMStart(llm, prompts) {
    const content = prompts.join("\n");
    // 写入到文件里面
    fs.appendFileSync(
      "./output.log",
      `[START] LLM 被调用，输入如下：\n${content}\n\n`
    );
  },
  handleLLMEnd(output) {
    const text = output.generations?.[0]?.[0].text; // 取出回复的内容
    fs.appendFileSync("./output.log", `[END] LLM 响应输出如下：\n${text}\n\n`);
  },
};
